#!/bin/bash

net=conv
epochs=640
while [ -n "$1" ]; do
    case "$1" in
    -net=*)
      net=${1#-net=}
      shift
      ;;
    -epochs=*)
      epochs=${1#-epochs=}
      shift
      ;;
    *)
      shift;;
  esac
done

for d in 1 2;
do
  for i in 0 1 2 3;
  do
    python3 train_softmax.py -data mnist -model "${net}" -dn ${d} -dt ${i} -e "${epochs}" || exit -1
  done
done
